package cn.itcast.web.shiro;

import cn.itcast.domain.system.Module;
import cn.itcast.domain.system.User;
import cn.itcast.service.system.ModuleService;
import cn.itcast.service.system.UserService;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * 自定义realm域
 *  继承AuthorizingRealm抽象父类
 *
 */

public class AuthRealm extends AuthorizingRealm {
    @Autowired
    private UserService userService;
    @Autowired
    private ModuleService moduleService;

    //授权方法
    /**
     *  参数： PrincipalCollection： 安全数据（用户） 集合
     *  返回值：所有操作权限
     *
     */
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
        //1获取当前用户（安全数据）
        User user = (User)principalCollection.getPrimaryPrincipal();
        //2.查询当前用户的所有模块
        List<Module> modules = moduleService.findByUserId(user.getId());
        //3.构造模块名称数组
        Set<String> stringPermission = new HashSet<String>();
        for (Module module : modules) {
            stringPermission.add(module.getName());
        }
        //4.构造返回值
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
        info.setStringPermissions(stringPermission);
        return info;
    }
    //认证方法

    /**
     * 参数： 用户登录的用户名和密码封装对象
     * 返回值：
     *          Authentication ：认证信息
     *
     */
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        //1.通过token转换为用户输入的邮箱和密码
        UsernamePasswordToken upToken = (UsernamePasswordToken) token;
        String email = upToken.getUsername();
        String password = new String(upToken.getPassword());
        //2.根据邮箱查询用户
        User user = userService.findByEmail(email);
        //3.判断用户是否存在
        if (user != null){
            //存在   ：构造AuthenticationInfo返回值
            /**
             * 参数：  1.安全数据（用户对象）
             *         2.密码（数据库密码）
             *         3.当前realm域的名称 （推荐：当前类名）
             */
          return  new SimpleAuthenticationInfo(user,user.getPassword(),this.getName());

        }else{
            //用户不存在
            return null;

        }

    }
}
